package Tree.TraversalByLevel;

import Tree.TreeNode;

import java.util.LinkedList;
import java.util.Queue;

public class ConnectNodeAndAddNext {
    public static void main(String[] args) {

    }
    public Node connect(Node root){
        Queue<Node> tmpQueue = new LinkedList<>();
        if(root == null) return null;
        tmpQueue.add(root);

        while(!tmpQueue.isEmpty()){
            int size = tmpQueue.size();
            Node prev = null;//存储前一个节点，初始化为null，后面会判断
            //开始层序遍历，每一次for遍历一层
            for(int i = 0; i < size; i++){
                Node cur = tmpQueue.poll();
                //如果不是第一个节点，将前一个节点的next指向当前节点
                if(prev != null) prev.next = cur;
                //如果是第一个节点，则将前一个节点更新为当前节点
                prev = cur;
                // 将当前节点的子节点加入队列
                if (cur.left != null) {
                    tmpQueue.add(cur.left);
                }
                if (cur.right != null) {
                    tmpQueue.add(cur.right);
                }
            }
            // 每层最后一个节点的 next 指向 null（可选，因为默认就是 null）
            if (prev != null) {
                prev.next = null;
            }
        }
        return root;
    }
}
